import torch
import torch.nn as nn
import sys
sys.path.append('.')

from models.common import autopad, Conv

class SimpleModule1(nn.Module):
    def __init__(self) -> None:
        '''
        简单的自定义 Module，用于导出测试
        '''
        super().__init__()
        self.cv1 = Conv(3, 64, 5, 2)
        self.cv2 = Conv(64, 128, 3, 2)

    def forward(self, x: 'torch.Tensor'):
        return self.cv2(self.cv1(x))